# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.

from maskrcnn_benchmark.utils.registry import Registry

BACKBONES = Registry()

RPN_HEADS = Registry()
STEREO_RPN_V1_HEADS = Registry()

ROI_BOX_FEATURE_EXTRACTORS = Registry()
ROI_BOX_PREDICTOR = Registry()

ROI_KEYPOINT_FEATURE_EXTRACTORS = Registry()
ROI_KEYPOINT_PREDICTOR = Registry()

ROI_MASK_FEATURE_EXTRACTORS = Registry()
ROI_MASK_PREDICTOR = Registry()

ROI_BOX_STEREO_V1_FEATURE_EXTRACTORS = Registry()
ROI_BOX_STEREO_V1_PREDICTOR = Registry()

ROI_BOX_STEREO_V2_FEATURE_EXTRACTORS = Registry()
ROI_BOX_STEREO_V2_PREDICTOR = Registry()

ROI_BOX_TD_V1_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V1_PREDICTOR = Registry()

ROI_BOX_TD_V2_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V2_PREDICTOR = Registry()

ROI_BOX_TD_V3_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V3_PREDICTOR = Registry()

ROI_BOX_TD_V4_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V4_PREDICTOR = Registry()

ROI_BOX_TD_V5_B1_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V5_B1_PREDICTOR = Registry()

ROI_BOX_TD_V5_B2_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V5_B2_PREDICTOR = Registry()

ROI_BOX_TD_V6_B1_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V6_B1_PREDICTOR = Registry()

ROI_BOX_TD_V6_B2_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V6_B2_PREDICTOR = Registry()

ROI_BOX_TD_V7_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V7_PREDICTOR = Registry()

ROI_BOX_TD_V8_FEATURE_EXTRACTORS = Registry()
ROI_BOX_TD_V8_PREDICTOR = Registry()
